hackerrank problem solving with python